{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "139002ef",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>ID</th>\n",
       "      <th>Day</th>\n",
       "      <th>Group</th>\n",
       "      <th>Originality</th>\n",
       "      <th>Interestingness</th>\n",
       "      <th>Writing</th>\n",
       "      <th>Coherence</th>\n",
       "      <th>Overall</th>\n",
       "      <th>Evaluator</th>\n",
       "      <th>unique_id</th>\n",
       "      <th>external</th>\n",
       "      <th>Humanlikeness</th>\n",
       "      <th>English</th>\n",
       "      <th>Experience</th>\n",
       "      <th>Ability</th>\n",
       "      <th>DAT</th>\n",
       "      <th>Total</th>\n",
       "      <th>Idea</th>\n",
       "      <th>Outline</th>\n",
       "      <th>Write</th>\n",
       "      <th>Edit</th>\n",
       "      <th>Satisfaction</th>\n",
       "      <th>Flexibility</th>\n",
       "      <th>Goal</th>\n",
       "      <th>Again</th>\n",
       "      <th>AI_Helpfulness</th>\n",
       "      <th>AI_Satisfaction</th>\n",
       "      <th>AI_Contribution</th>\n",
       "      <th>DAT_Group</th>\n",
       "      <th>Group_</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>5</td>\n",
       "      <td>3</td>\n",
       "      <td>4</td>\n",
       "      <td>7</td>\n",
       "      <td>4.0</td>\n",
       "      <td>Benjamin Joers</td>\n",
       "      <td>1_1</td>\n",
       "      <td>0</td>\n",
       "      <td>4.5</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>86.81</td>\n",
       "      <td>80.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>1</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>5</td>\n",
       "      <td>6</td>\n",
       "      <td>4</td>\n",
       "      <td>5</td>\n",
       "      <td>5.0</td>\n",
       "      <td>Rio Dharma</td>\n",
       "      <td>1_1</td>\n",
       "      <td>0</td>\n",
       "      <td>4.5</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>86.81</td>\n",
       "      <td>80.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>8.0</td>\n",
       "      <td>55.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>3</td>\n",
       "      <td>5</td>\n",
       "      <td>2.0</td>\n",
       "      <td>allison liegner</td>\n",
       "      <td>1_2</td>\n",
       "      <td>0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>86.81</td>\n",
       "      <td>85.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1</td>\n",
       "      <td>2</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>6</td>\n",
       "      <td>7</td>\n",
       "      <td>7</td>\n",
       "      <td>6</td>\n",
       "      <td>7.0</td>\n",
       "      <td>Ryan Ho</td>\n",
       "      <td>1_2</td>\n",
       "      <td>0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>86.81</td>\n",
       "      <td>85.0</td>\n",
       "      <td>10.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>15.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>70.0</td>\n",
       "      <td>3</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "      <td>Human Confirmation</td>\n",
       "      <td>6</td>\n",
       "      <td>5</td>\n",
       "      <td>6</td>\n",
       "      <td>6</td>\n",
       "      <td>6.0</td>\n",
       "      <td>Nathan Kidambi</td>\n",
       "      <td>2_1</td>\n",
       "      <td>0</td>\n",
       "      <td>6.5</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>89.58</td>\n",
       "      <td>NaN</td>\n",
       "      <td>45.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2191</th>\n",
       "      <td>294</td>\n",
       "      <td>2</td>\n",
       "      <td>Copilot</td>\n",
       "      <td>6</td>\n",
       "      <td>4</td>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>5.0</td>\n",
       "      <td>Pema Euden</td>\n",
       "      <td>294_2</td>\n",
       "      <td>1</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>87.08</td>\n",
       "      <td>45.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>12.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>7.0</td>\n",
       "      <td>91.0</td>\n",
       "      <td>3</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2192</th>\n",
       "      <td>295</td>\n",
       "      <td>1</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>5</td>\n",
       "      <td>4</td>\n",
       "      <td>4</td>\n",
       "      <td>4</td>\n",
       "      <td>4.0</td>\n",
       "      <td>Pema Euden</td>\n",
       "      <td>295_1</td>\n",
       "      <td>1</td>\n",
       "      <td>6.5</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>74.63</td>\n",
       "      <td>66.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2193</th>\n",
       "      <td>295</td>\n",
       "      <td>1</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>6</td>\n",
       "      <td>5</td>\n",
       "      <td>5.0</td>\n",
       "      <td>Alan Wu</td>\n",
       "      <td>295_1</td>\n",
       "      <td>1</td>\n",
       "      <td>6.5</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>74.63</td>\n",
       "      <td>66.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>40.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>2.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2194</th>\n",
       "      <td>295</td>\n",
       "      <td>2</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>6</td>\n",
       "      <td>5.0</td>\n",
       "      <td>Pema Euden</td>\n",
       "      <td>295_2</td>\n",
       "      <td>1</td>\n",
       "      <td>3.5</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>74.63</td>\n",
       "      <td>55.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>71.0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2195</th>\n",
       "      <td>295</td>\n",
       "      <td>2</td>\n",
       "      <td>Human Creativity</td>\n",
       "      <td>5</td>\n",
       "      <td>5</td>\n",
       "      <td>7</td>\n",
       "      <td>7</td>\n",
       "      <td>6.0</td>\n",
       "      <td>Ziqi Yang</td>\n",
       "      <td>295_2</td>\n",
       "      <td>1</td>\n",
       "      <td>3.5</td>\n",
       "      <td>2.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>74.63</td>\n",
       "      <td>55.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>20.0</td>\n",
       "      <td>25.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>3.0</td>\n",
       "      <td>4.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>1.0</td>\n",
       "      <td>6.0</td>\n",
       "      <td>5.0</td>\n",
       "      <td>71.0</td>\n",
       "      <td>2</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>2196 rows × 30 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       ID  Day               Group  Originality  Interestingness  Writing  \\\n",
       "0       1    1    Human Creativity            5                3        4   \n",
       "1       1    1    Human Creativity            5                6        4   \n",
       "2       1    2    Human Creativity            2                1        3   \n",
       "3       1    2    Human Creativity            6                7        7   \n",
       "4       2    1  Human Confirmation            6                5        6   \n",
       "...   ...  ...                 ...          ...              ...      ...   \n",
       "2191  294    2             Copilot            6                4        5   \n",
       "2192  295    1    Human Creativity            5                4        4   \n",
       "2193  295    1    Human Creativity            5                5        6   \n",
       "2194  295    2    Human Creativity            5                5        5   \n",
       "2195  295    2    Human Creativity            5                5        7   \n",
       "\n",
       "      Coherence  Overall        Evaluator unique_id  external  Humanlikeness  \\\n",
       "0             7      4.0   Benjamin Joers       1_1         0            4.5   \n",
       "1             5      5.0       Rio Dharma       1_1         0            4.5   \n",
       "2             5      2.0  allison liegner       1_2         0            3.0   \n",
       "3             6      7.0          Ryan Ho       1_2         0            3.0   \n",
       "4             6      6.0   Nathan Kidambi       2_1         0            6.5   \n",
       "...         ...      ...              ...       ...       ...            ...   \n",
       "2191          5      5.0       Pema Euden     294_2         1            3.0   \n",
       "2192          4      4.0       Pema Euden     295_1         1            6.5   \n",
       "2193          5      5.0          Alan Wu     295_1         1            6.5   \n",
       "2194          6      5.0       Pema Euden     295_2         1            3.5   \n",
       "2195          7      6.0        Ziqi Yang     295_2         1            3.5   \n",
       "\n",
       "      English  Experience  Ability    DAT  Total  Idea  Outline  Write  Edit  \\\n",
       "0         3.0         1.0      2.0  86.81   80.0  12.0      8.0   55.0   5.0   \n",
       "1         3.0         1.0      2.0  86.81   80.0  12.0      8.0   55.0   5.0   \n",
       "2         3.0         1.0      2.0  86.81   85.0  10.0     20.0   15.0  40.0   \n",
       "3         3.0         1.0      2.0  86.81   85.0  10.0     20.0   15.0  40.0   \n",
       "4         2.0         1.0      1.0  89.58    NaN  45.0      NaN    NaN   NaN   \n",
       "...       ...         ...      ...    ...    ...   ...      ...    ...   ...   \n",
       "2191      3.0         3.0      2.0  87.08   45.0   3.0      5.0   25.0  12.0   \n",
       "2192      2.0         1.0      1.0  74.63   66.0   1.0      5.0   40.0  20.0   \n",
       "2193      2.0         1.0      1.0  74.63   66.0   1.0      5.0   40.0  20.0   \n",
       "2194      2.0         1.0      1.0  74.63   55.0   5.0     20.0   25.0   5.0   \n",
       "2195      2.0         1.0      1.0  74.63   55.0   5.0     20.0   25.0   5.0   \n",
       "\n",
       "      Satisfaction  Flexibility  Goal  Again  AI_Helpfulness  AI_Satisfaction  \\\n",
       "0              1.0          4.0   2.0    0.0             NaN              NaN   \n",
       "1              1.0          4.0   2.0    0.0             NaN              NaN   \n",
       "2              3.0          3.0   3.0    0.0             4.0              5.0   \n",
       "3              3.0          3.0   3.0    0.0             4.0              5.0   \n",
       "4              2.0          6.0   2.0    0.0             NaN              NaN   \n",
       "...            ...          ...   ...    ...             ...              ...   \n",
       "2191           7.0          6.0   5.0    1.0             7.0              7.0   \n",
       "2192           5.0          4.0   2.0    0.0             NaN              NaN   \n",
       "2193           5.0          4.0   2.0    0.0             NaN              NaN   \n",
       "2194           3.0          4.0   5.0    1.0             6.0              5.0   \n",
       "2195           3.0          4.0   5.0    1.0             6.0              5.0   \n",
       "\n",
       "      AI_Contribution  DAT_Group  Group_  \n",
       "0                 NaN          3       1  \n",
       "1                 NaN          3       1  \n",
       "2                70.0          3       1  \n",
       "3                70.0          3       1  \n",
       "4                 NaN          3       2  \n",
       "...               ...        ...     ...  \n",
       "2191             91.0          3       3  \n",
       "2192              NaN          2       1  \n",
       "2193              NaN          2       1  \n",
       "2194             71.0          2       1  \n",
       "2195             71.0          2       1  \n",
       "\n",
       "[2196 rows x 30 columns]"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import pandas as pd\n",
    "import numpy as np\n",
    "import matplotlib.pyplot as plt\n",
    "import matplotlib as mpl\n",
    "import matplotlib.transforms as transforms\n",
    "import seaborn as sns\n",
    "from IPython.display import set_matplotlib_formats\n",
    "%matplotlib inline\n",
    "set_matplotlib_formats('retina')\n",
    "from scipy import stats\n",
    "import statsmodels.formula.api as smf\n",
    "import statsmodels.api as sm\n",
    "import warnings\n",
    "import matplotlib.path as mpath\n",
    "sns.set(rc={\"figure.dpi\":100, 'savefig.dpi':300})\n",
    "sns.set_context('notebook')\n",
    "sns.set_style(\"ticks\")\n",
    "warnings.filterwarnings('ignore')\n",
    "pd.set_option('display.max_columns', None)\n",
    "df = pd.read_csv('Replication Data for Designing Human and Generative AI Collaboration.csv').iloc[:,1:]\n",
    "df"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "cd894976",
   "metadata": {},
   "source": [
    "# Table 1. Descriptive Stats"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "68c1f247",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Overall</th>\n",
       "      <th>Originality</th>\n",
       "      <th>Interestingness</th>\n",
       "      <th>Writing</th>\n",
       "      <th>Coherence</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>count</th>\n",
       "      <td>2196.000000</td>\n",
       "      <td>2196.000000</td>\n",
       "      <td>2196.000000</td>\n",
       "      <td>2196.000000</td>\n",
       "      <td>2196.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>mean</th>\n",
       "      <td>4.907104</td>\n",
       "      <td>4.971311</td>\n",
       "      <td>4.865665</td>\n",
       "      <td>5.209472</td>\n",
       "      <td>5.197632</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>std</th>\n",
       "      <td>1.352198</td>\n",
       "      <td>1.380171</td>\n",
       "      <td>1.464752</td>\n",
       "      <td>1.366978</td>\n",
       "      <td>1.356034</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>min</th>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25%</th>\n",
       "      <td>4.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>4.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50%</th>\n",
       "      <td>5.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>5.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>75%</th>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>max</th>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "           Overall  Originality  Interestingness      Writing    Coherence\n",
       "count  2196.000000  2196.000000      2196.000000  2196.000000  2196.000000\n",
       "mean      4.907104     4.971311         4.865665     5.209472     5.197632\n",
       "std       1.352198     1.380171         1.464752     1.366978     1.356034\n",
       "min       1.000000     1.000000         1.000000     1.000000     1.000000\n",
       "25%       4.000000     4.000000         4.000000     4.000000     4.000000\n",
       "50%       5.000000     5.000000         5.000000     5.000000     5.000000\n",
       "75%       6.000000     6.000000         6.000000     6.000000     6.000000\n",
       "max       7.000000     7.000000         7.000000     7.000000     7.000000"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df[['Overall', 'Originality', 'Interestingness', 'Writing', 'Coherence']].describe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "id": "5f0ffa1a",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>ID</th>\n",
       "      <th>DAT</th>\n",
       "      <th>Experience</th>\n",
       "      <th>Ability</th>\n",
       "      <th>English</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>count</th>\n",
       "      <td>285.000000</td>\n",
       "      <td>280.000000</td>\n",
       "      <td>280.000000</td>\n",
       "      <td>280.000000</td>\n",
       "      <td>280.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>mean</th>\n",
       "      <td>149.771930</td>\n",
       "      <td>82.279107</td>\n",
       "      <td>1.621429</td>\n",
       "      <td>1.896429</td>\n",
       "      <td>2.767857</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>std</th>\n",
       "      <td>85.145653</td>\n",
       "      <td>7.126907</td>\n",
       "      <td>0.713092</td>\n",
       "      <td>0.672012</td>\n",
       "      <td>0.431347</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>min</th>\n",
       "      <td>1.000000</td>\n",
       "      <td>29.330000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25%</th>\n",
       "      <td>77.000000</td>\n",
       "      <td>78.422500</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>3.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50%</th>\n",
       "      <td>151.000000</td>\n",
       "      <td>82.090000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>3.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>75%</th>\n",
       "      <td>223.000000</td>\n",
       "      <td>87.365000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>3.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>max</th>\n",
       "      <td>295.000000</td>\n",
       "      <td>95.280000</td>\n",
       "      <td>3.000000</td>\n",
       "      <td>3.000000</td>\n",
       "      <td>3.000000</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               ID         DAT  Experience     Ability     English\n",
       "count  285.000000  280.000000  280.000000  280.000000  280.000000\n",
       "mean   149.771930   82.279107    1.621429    1.896429    2.767857\n",
       "std     85.145653    7.126907    0.713092    0.672012    0.431347\n",
       "min      1.000000   29.330000    1.000000    1.000000    1.000000\n",
       "25%     77.000000   78.422500    1.000000    1.000000    3.000000\n",
       "50%    151.000000   82.090000    1.000000    2.000000    3.000000\n",
       "75%    223.000000   87.365000    2.000000    2.000000    3.000000\n",
       "max    295.000000   95.280000    3.000000    3.000000    3.000000"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.groupby(by=['ID'])[['DAT', 'Experience', 'Ability', 'English']].first().reset_index().describe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "id": "4dc16e8b",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>ID</th>\n",
       "      <th>Day</th>\n",
       "      <th>Satisfaction</th>\n",
       "      <th>Flexibility</th>\n",
       "      <th>Goal</th>\n",
       "      <th>Again</th>\n",
       "      <th>AI_Helpfulness</th>\n",
       "      <th>AI_Satisfaction</th>\n",
       "      <th>AI_Contribution</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>count</th>\n",
       "      <td>549.000000</td>\n",
       "      <td>549.000000</td>\n",
       "      <td>508.000000</td>\n",
       "      <td>508.000000</td>\n",
       "      <td>508.000000</td>\n",
       "      <td>508.000000</td>\n",
       "      <td>248.000000</td>\n",
       "      <td>248.000000</td>\n",
       "      <td>248.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>mean</th>\n",
       "      <td>150.604736</td>\n",
       "      <td>1.493625</td>\n",
       "      <td>4.901575</td>\n",
       "      <td>4.950787</td>\n",
       "      <td>4.728346</td>\n",
       "      <td>0.718504</td>\n",
       "      <td>5.471774</td>\n",
       "      <td>5.225806</td>\n",
       "      <td>76.254032</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>std</th>\n",
       "      <td>84.825984</td>\n",
       "      <td>0.500415</td>\n",
       "      <td>1.527576</td>\n",
       "      <td>1.474149</td>\n",
       "      <td>1.574270</td>\n",
       "      <td>0.450172</td>\n",
       "      <td>1.422495</td>\n",
       "      <td>1.646189</td>\n",
       "      <td>17.077698</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>min</th>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>21.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>25%</th>\n",
       "      <td>79.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>0.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>4.000000</td>\n",
       "      <td>70.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>50%</th>\n",
       "      <td>152.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>5.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>80.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>75%</th>\n",
       "      <td>224.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>6.000000</td>\n",
       "      <td>90.000000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>max</th>\n",
       "      <td>295.000000</td>\n",
       "      <td>2.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>1.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>7.000000</td>\n",
       "      <td>100.000000</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "               ID         Day  Satisfaction  Flexibility        Goal  \\\n",
       "count  549.000000  549.000000    508.000000   508.000000  508.000000   \n",
       "mean   150.604736    1.493625      4.901575     4.950787    4.728346   \n",
       "std     84.825984    0.500415      1.527576     1.474149    1.574270   \n",
       "min      1.000000    1.000000      1.000000     1.000000    1.000000   \n",
       "25%     79.000000    1.000000      4.000000     4.000000    4.000000   \n",
       "50%    152.000000    1.000000      5.000000     5.000000    5.000000   \n",
       "75%    224.000000    2.000000      6.000000     6.000000    6.000000   \n",
       "max    295.000000    2.000000      7.000000     7.000000    7.000000   \n",
       "\n",
       "            Again  AI_Helpfulness  AI_Satisfaction  AI_Contribution  \n",
       "count  508.000000      248.000000       248.000000       248.000000  \n",
       "mean     0.718504        5.471774         5.225806        76.254032  \n",
       "std      0.450172        1.422495         1.646189        17.077698  \n",
       "min      0.000000        1.000000         1.000000        21.000000  \n",
       "25%      0.000000        5.000000         4.000000        70.000000  \n",
       "50%      1.000000        6.000000         6.000000        80.000000  \n",
       "75%      1.000000        6.000000         6.000000        90.000000  \n",
       "max      1.000000        7.000000         7.000000       100.000000  "
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.groupby(by=['ID', 'Day'])[['Satisfaction', 'Flexibility', 'Goal', 'Again', 'AI_Helpfulness', 'AI_Satisfaction', 'AI_Contribution']].first().reset_index().describe()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "cbc42dab",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.9.16"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}
